package com.xianggu.demo;

import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

/**
 * 求集合的并集和交集
 */
public class SetOperations {
    public static void main(String[] args) {
        List<Integer> list1 = List.of(1, 2, 3, 4, 5);
        List<Integer> list2 = List.of(3, 4, 5, 6, 7);
        
        // 求并集
        Set<Integer> union = new HashSet<>(list1);
        union.addAll(list2);
        System.out.println("并集: " + union);
        
        // 求交集
        Set<Integer> intersection = new HashSet<>(list1);
        intersection.retainAll(list2);

        Set<Integer> intersection_2 = new HashSet<>();
        for (Integer i : list1) {
            if (list2.contains(i)) {
                intersection_2.add(i);
            }
        }

        System.out.println("交集: " + intersection);
        System.out.println("交集: " + intersection_2);
    }
    
    // 通用方法版本
    public static <T> List<T> getUnion(List<T> list1, List<T> list2) {
        Set<T> set = new HashSet<>(list1);
        set.addAll(list2);
        return new ArrayList<>(set);
    }
    
    public static <T> List<T> getIntersection(List<T> list1, List<T> list2) {
        Set<T> set = new HashSet<>(list1);
        set.retainAll(list2);
        return new ArrayList<>(set);
    }
}